import os
from dotenv import load_dotenv

# 加载环境变量
load_dotenv()

# 智谱 GLM-4 配置
GLM_CONFIG = {
    "api_key": os.getenv("ZHIPU_API_KEY"),
    "model": "glm-4",  # 使用 GLM-4 模型
    "temperature": 0.95,  # 默认值
    "top_p": 0.7,  # 默认值
    "max_tokens": 1024,  # 默认值
}

# 微信配置
WECHAT_CONFIG = {
    "auto_reply_all": False,  # 是否自动回复所有消息
    "group_reply_mention": True,  # 群聊是否只回复@消息
    "blocked_users": [],  # 不回复的用户列表
    "blocked_groups": [],  # 不回复的群组列表
    "reply_delay": 1,  # 回复延迟（秒）
}

# 日志配置
LOG_CONFIG = {
    "level": "INFO",
    "format": "<green>{time:YYYY-MM-DD HH:mm:ss}</green> | <level>{level: <8}</level> | <cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>",
    "rotation": "1 day",
} 